from lib import SparkApi
#以下密钥信息从控制台获取
appid = "535f4ac0"  #填写控制台中获取的 APPID 信息

api_secret = "ZGQwZTYxNWRkZjJjZDJlZTNkOTI5ZjUz"#填写控制台中获取的 APISecret 信息

api_key ="2bd3742e722f8f1c23ed37dff0112ec5"#填写控制台中获取的 APIKey 信息


#用于配置大模型版本，默认“general/generalv2”
# domain = "general"   # v1.5版本
# domain = "generalv2"    # v2.0版本
domain = "generalv3" 
#云端环境的服务地址
#Spark_url = "ws://spark-api.xf-yun.com/v1.1/chat"  # v1.5环境的地址
# Spark_url = "ws://spark-api.xf-yun.com/v2.1/chat"  # v2.0环境的地址
Spark_url = "ws://spark-api.xf-yun.com/v3.1/chat"

text =[]

# length = 0

def getText(role,content):
    jsoncon = {}
    jsoncon["role"] = role
    jsoncon["content"] = content
    text.append(jsoncon)
    return text

def getlength(text):
    length = 0
    for content in text:
        temp = content["content"]
        leng = len(temp)
        length += leng
    return length

def checklen(text):
    while (getlength(text) > 8000):
        del text[0]
    return text


with open('test_data/test_qa.txt') as f:
    input = f.read()
question = checklen(getText("user",input))
print(question)
SparkApi.answer =""
SparkApi.main(appid,api_key,api_secret,Spark_url,domain,question)
SparkApi.answer =""
print(SparkApi.answer)



